package kit.docker.math;

import java.math.BigDecimal;
import java.text.DecimalFormat;

public class MathKit {

    // 默认除法运算精度
    private static final int DEFAULT_DIV_SCALE = 2;
    // 默认格式化精度
    private static final int DEFAULT_FORMAT_SCALE = 2;

    /**
     * 精确加法运算
     */
    public static double add(double v1, double v2) {
        BigDecimal b1 = new BigDecimal(Double.toString(v1));
        BigDecimal b2 = new BigDecimal(Double.toString(v2));
        return b1.add(b2).doubleValue();
    }

    /**
     * 精确加法运算
     */
    public static String add(String v1, String v2) {
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return format(b1.add(b2).doubleValue());
    }

    public static String add(String v1, String v2, int scale) {
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return format(b1.add(b2).doubleValue(), scale);
    }

    /**
     * 精确减法运算
     */
    public static double subtract(double v1, double v2) {
        BigDecimal b1 = new BigDecimal(Double.toString(v1));
        BigDecimal b2 = new BigDecimal(Double.toString(v2));
        return b1.subtract(b2).doubleValue();
    }

    /**
     * 精确减法运算
     */
    public static String subtract(String v1, String v2) {
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return format(b1.subtract(b2).doubleValue());
    }

    public static String subtract(String v1, String v2, int scale) {
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return format(b1.subtract(b2).doubleValue(), scale);
    }

    /**
     * 精确乘法运算
     */
    public static double multiply(double v1, double v2) {
        BigDecimal b1 = new BigDecimal(Double.toString(v1));
        BigDecimal b2 = new BigDecimal(Double.toString(v2));
        return b1.multiply(b2).doubleValue();
    }

    /**
     * 精确乘法运算
     */
    public static String multiply(String v1, String v2) {
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return format(b1.multiply(b2).doubleValue());
    }

    public static String multiply(String v1, String v2, int scale) {
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return format(b1.multiply(b2).doubleValue(), scale);
    }

    /**
     * 相对精确的除法运算
     * 当发生除不尽的情况时，精确到小数点后10位，
     * 以后的数字四舍五入，舍入模式采用 ROUND_HALF_EVEN
     */
    public static double divide(double v1, double v2) {
        return divide(v1, v2, DEFAULT_DIV_SCALE);
    }

    /**
     * 相对精确的除法运算
     * 当发生除不尽的情况时，由scale参数指定精度，
     * 以后的数字四舍五入，舍入模式采用 ROUND_HALF_EVEN
     */
    public static double divide(double v1, double v2, int scale) {
        return divide(v1, v2, scale, BigDecimal.ROUND_HALF_EVEN);
    }

    /**
     * 相对精确的除法运算
     * 当发生除不尽的情况时，由scale参数指定精度，
     * 以后的数字四舍五入，舍入模式采用用户指定round_mode舍入模式
     */
    public static double divide(double v1, double v2, int scale, int round_mode) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        BigDecimal b1 = new BigDecimal(Double.toString(v1));
        BigDecimal b2 = new BigDecimal(Double.toString(v2));
        return b1.divide(b2, scale, round_mode).doubleValue();
    }

    /**
     * 相对精确的除法运算
     * 当发生除不尽的情况时，精确到小数点后10位，
     * 以后的数字四舍五入，舍入模式采用 ROUND_HALF_EVEN
     */
    public static String divide(String v1, String v2) {
        return divide(v1, v2, DEFAULT_DIV_SCALE);
    }

    /**
     * 相对精确的除法运算
     * 当发生除不尽的情况时，由scale参数指定精度，
     * 以后的数字四舍五入，舍入模式采用 ROUND_HALF_EVEN
     */
    public static String divide(String v1, String v2, int scale) {
        return divide(v1, v2, DEFAULT_DIV_SCALE, BigDecimal.ROUND_HALF_EVEN);
    }

    /**
     * 相对精确的除法运算
     * 当发生除不尽的情况时，由scale参数指定精度，
     * 以后的数字四舍五入，舍入模式采用用户指定round_mode舍入模式
     */
    public static String divide(String v1, String v2, int scale, int round_mode) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        BigDecimal b1 = new BigDecimal(v1);
        BigDecimal b2 = new BigDecimal(v2);
        return b1.divide(b2, scale, round_mode).toString();
    }

    /**
     * 提供精确的小数位四舍五入处理,舍入模式采用ROUND_HALF_EVEN
     */
    public static double round(double v, int scale) {
        return round(v, scale, BigDecimal.ROUND_HALF_EVEN);
    }

    /**
     * 提供精确的小数位四舍五入处理
     */

    public static double round(double v, int scale, int round_mode) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        BigDecimal b = new BigDecimal(Double.toString(v));
        return b.setScale(scale, round_mode).doubleValue();
    }

    /**
     * 提供精确的小数位四舍五入处理,舍入模式采用ROUND_HALF_EVEN
     */

    public static String round(String v, int scale) {
        return round(v, scale, BigDecimal.ROUND_HALF_EVEN);
    }

    /**
     * 提供精确的小数位四舍五入处理
     */
    public static String round(String v, int scale, int round_mode) {
        if (scale < 0) {
            throw new IllegalArgumentException("The scale must be a positive integer or zero");
        }
        BigDecimal b = new BigDecimal(v);
        return b.setScale(scale, round_mode).toString();
    }

    // =================================================================================

    public static String format(double value) {
        return format(value, DEFAULT_FORMAT_SCALE);
    }

    public static String format(double value, int afterLen) {
        if (afterLen < 0) {
            throw new RuntimeException("保留小数点位数应大于0");
        }
        StringBuilder formatStr = new StringBuilder("#" + (afterLen == 0 ? "" : "."));
        for (int i = 1; i <= afterLen; i++) {
            formatStr.append("0");
        }
        DecimalFormat df = new DecimalFormat(formatStr.toString());
        return df.format(value);
    }

    public static String format(String value) {
        return format(Double.valueOf(value));
    }

    public static String format(String value, int afterLen) {
        return format(Double.valueOf(value), afterLen);
    }

}
